# 导包
import unittest
import app
from api.login import LoginAPI
from utils import coommon_assert

# 创建测试类
class TestLoginAPI(unittest.TestCase):
    # 前置处理
    def setUp(self):
        # 实例化接口类
        self.login_api = LoginAPI()
    # 登录成功
    def test01_login_success(self):
        # 获取登录响应
        login_data = {
            "mobile": "13800000002",
            "password": "123456"
        }
        response = self.login_api.login(login_data)
        print(response.json())

        # 断言
        self.assertEqual(200, response.status_code)
        self.assertEqual(True, response.json().get("success"))
        self.assertEqual(10000, response.json().get("code"))
        self.assertIn("操作成功", response.json().get("message"))
        # 调用公共断言方法实现断言
        coommon_assert(self, response)

        # 提取登录接口响应中的token信息，解决员工管理接口的依赖
        app.header_data["Authorization"] = "Bearer " + response.json().get("data")
        print(app.header_data["Authorization"])


